from django.contrib import messages
from django.shortcuts import redirect
import functools


def flash(request, title, text, level='info'):
    """
    利用django的message系统发送一个信息。
    """
    level_map = {
        'info': messages.INFO,
        'debug': messages.DEBUG,
        'success': messages.SUCCESS,
        'warning': messages.WARNING,
        'error': messages.ERROR
    }

    level = level_map[level]

    messages.add_message(request, level, text, extra_tags=title)
    return 1


def login_required(f):

  def wrapped_f(request):
    if not request.session.get('user_id'):
      return redirect('inner_home')
    return f(request)
  return wrapped_f
